package com.jingxi.common.util;

import org.apache.commons.net.ftp.FTPClient;
import org.apache.commons.net.ftp.FTPFile;
import org.apache.commons.net.ftp.FTPHTTPClient;
import org.apache.commons.net.ftp.FTPReply;

import java.io.*;

public class FtpUtil {

	/*
	 * Description: 向FTP服务器上传文件
	 * 
	 * @paramhost FTP服务器hostname
	 * 
	 * @paramport FTP服务器端口
	 * 
	 * @paramusername FTP登录账号
	 * 
	 * @parampassword FTP登录密码
	 * 
	 * @parambasePath FTP服务器基础目录
	 * 
	 * @paramfilePath FTP服务器文件存放路径。例如分日期存放：/2017/01/01。文件的路径为basePath+filePath
	 * 
	 * @paramfilename 上传到FTP服务器上的文件名
	 * 
	 * @paraminput 输入流
	 * 
	 * @return成功返回true，否则返回false
	 */

	public static boolean uploadFile(String host, int port, String username, String password, String basePath,
			String filePath, String filename, InputStream input) throws Exception {
		boolean result = false;

		FTPClient ftp = new FTPClient();

		int reply;

		ftp.connect(host, port);// 连接FTP服务器
		// 如果采用默认端口，可以使用ftp.connect(host)的方式直接连接FTP服务器

		ftp.login(username, password);// 登录

		reply = ftp.getReplyCode();

		ftp.enterLocalPassiveMode();
		
		if (!FTPReply.isPositiveCompletion(reply)) {
			ftp.disconnect();

			return result;
		}
		// 切换到上传目录
		if (!ftp.changeWorkingDirectory(basePath + filePath)) {
			// 如果目录不存在创建目录
			String[] dirs = filePath.split("/");

			String tempPath = basePath;

			for (String dir : dirs) {
				if (null == dir || "".equals(dir))
					continue;
				tempPath += "/" + dir;
				if (!ftp.changeWorkingDirectory(tempPath)) {
					if (!ftp.makeDirectory(tempPath)) {
						return result;
					} else {
						ftp.changeWorkingDirectory(tempPath);
					}
				}
			}
		}

		// 设置上传文件的类型为二进制类型

		ftp.setFileType(FTPClient.BINARY_FILE_TYPE);

		// 上传文件

		if (!ftp.storeFile(filename, input)) {
			System.out.println("return false 2");
			return result;
		}

		input.close();

		ftp.logout();

		result = true;
		if (ftp.isConnected()) {
				ftp.disconnect();
		}
		return result;
	}

	/*
	 * Description: 从FTP服务器下载文件
	 * 
	 * @paramhost FTP服务器hostname
	 * 
	 * @paramport FTP服务器端口
	 * 
	 * @paramusername FTP登录账号
	 * 
	 * @parampassword FTP登录密码
	 * 
	 * @paramremotePath FTP服务器上的相对路径
	 * 
	 * @paramfileName 要下载的文件名
	 * 
	 * @paramlocalPath 下载后保存到本地的路径
	 * 
	 * @return
	 */
	public static boolean downloadFile(String host, int port, String username, String password, String remotePath,
			String fileName, String localPath) {
		boolean result = false;
		FTPClient ftp = new FTPClient();

		try {
			int reply;

			ftp.connect(host, port);
			// 如果采用默认端口，可以使用ftp.connect(host)的方式直接连接FTP服务器

			ftp.login(username, password);

			reply = ftp.getReplyCode();

			if (!FTPReply.isPositiveCompletion(reply)) {
				ftp.disconnect();
				return result;
			}

			ftp.changeWorkingDirectory(remotePath);// 转移到FTP服务器目录

			FTPFile[] fs = ftp.listFiles();

			for (FTPFile ff : fs) {

				if (ff.getName().equals(fileName)) {
					File localFile = new File(localPath + "/" + ff.getName());

					OutputStream is = new FileOutputStream(localFile);

					ftp.retrieveFile(ff.getName(), is);

					is.close();
				}

			}

			ftp.logout();
			result = true;
		} catch (IOException e) {
			e.printStackTrace();
		} finally {
			if (ftp.isConnected()) {
				try {
					ftp.disconnect();
				} catch (IOException ioe) {
				}
			}
		}

		return result;
	}
}
